/*
 * Copyright 2012 Faculty of Informatics - Masaryk University.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package cz.muni.fi.pa165.creatures.dto.web.mapping;

import cz.muni.fi.pa165.creatures.dto.mapping.api.MappingDTO;
import cz.muni.fi.pa165.creatures.dto.web.PersonFunctionDTO;
import cz.muni.fi.pa165.creatures.model.web.PersonFunction;
import cz.muni.fi.pa165.creatures.model.web.PersonFunctionPK;

/**
 *
 * @author smikloso
 */
public class PersonFunctionMapping implements
        MappingDTO<PersonFunctionDTO, PersonFunction> {

    @Override
    public PersonFunctionDTO entityToDTO(PersonFunction entity) {
        if (entity == null || entity.getPersonFunctionPK() == null) {
            return null;
        }

        PersonFunctionDTO dto =
                new PersonFunctionDTO(
                    entity.getPersonFunctionPK().getPersonId().toString(),
                    entity.getPersonFunctionPK().getFunctionId().toString());

        return dto;
    }

    @Override
    public PersonFunction DTOtoEntity(PersonFunctionDTO dto) {
        if (dto == null) {
            return null;
        }

        if (dto.getFunctionId() == null
                || dto.getPersonId() == null) {
            return null;
        }

        PersonFunction pf = new PersonFunction();

        PersonFunctionPK pk = new PersonFunctionPK();

        if (dto.getPersonId() != null) {
            pk.setPersonId(Long.parseLong(dto.getPersonId()));
        }

        if (dto.getFunctionId() != null) {
            pk.setFunctionId(Long.parseLong(dto.getFunctionId()));
        }

        pf.setPersonFunctionPK(pk);

        return pf;
    }
}
